home *** CD-ROM | disk | FTP | other *** search
/ Chip 2007 January, February, March & April / Chip-Cover-CD-2007-02.iso / Pakiet bezpieczenstwa / mini Pentoo LiveCD 2006.1 / mpentoo-2006.1.iso / livecd.squashfs / usr / lib / python2.4 / ihooks.pyo (.txt) < prev    next >
Python Compiled Bytecode  |  2005-10-18  |  20KB  |  661 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyo (Python 2.4)
  3.  
  4. '''Import hook support.
  5.  
  6. Consistent use of this module will make it possible to change the
  7. different mechanisms involved in loading modules independently.
  8.  
  9. While the built-in module imp exports interfaces to the built-in
  10. module searching and loading algorithm, and it is possible to replace
  11. the built-in function __import__ in order to change the semantics of
  12. the import statement, until now it has been difficult to combine the
  13. effect of different __import__ hacks, like loading modules from URLs
  14. by rimport.py, or restricted execution by rexec.py.
  15.  
  16. This module defines three new concepts:
  17.  
  18. 1) A "file system hooks" class provides an interface to a filesystem.
  19.  
  20. One hooks class is defined (Hooks), which uses the interface provided
  21. by standard modules os and os.path.  It should be used as the base
  22. class for other hooks classes.
  23.  
  24. 2) A "module loader" class provides an interface to search for a
  25. module in a search path and to load it.  It defines a method which
  26. searches for a module in a single directory; by overriding this method
  27. one can redefine the details of the search.  If the directory is None,
  28. built-in and frozen modules are searched instead.
  29.  
  30. Two module loader class are defined, both implementing the search
  31. strategy used by the built-in __import__ function: ModuleLoader uses
  32. the imp module\'s find_module interface, while HookableModuleLoader
  33. uses a file system hooks class to interact with the file system.  Both
  34. use the imp module\'s load_* interfaces to actually load the module.
  35.  
  36. 3) A "module importer" class provides an interface to import a
  37. module, as well as interfaces to reload and unload a module.  It also
  38. provides interfaces to install and uninstall itself instead of the
  39. default __import__ and reload (and unload) functions.
  40.  
  41. One module importer class is defined (ModuleImporter), which uses a
  42. module loader instance passed in (by default HookableModuleLoader is
  43. instantiated).
  44.  
  45. The classes defined here should be used as base classes for extended
  46. functionality along those lines.
  47.  
  48. If a module importer class supports dotted names, its import_module()
  49. must return a different value depending on whether it is called on
  50. behalf of a "from ... import ..." statement or not.  (This is caused
  51. by the way the __import__ hook is used by the Python interpreter.)  It
  52. would also do wise to install a different version of reload().
  53.  
  54. '''
  55. import __builtin__
  56. import imp
  57. import os
  58. import sys
  59. __all__ = [
  60.     'BasicModuleLoader',
  61.     'Hooks',
  62.     'ModuleLoader',
  63.     'FancyModuleLoader',
  64.     'BasicModuleImporter',
  65.     'ModuleImporter',
  66.     'install',
  67.     'uninstall']
  68. VERBOSE = 0
  69. from imp import C_EXTENSION, PY_SOURCE, PY_COMPILED
  70. from imp import C_BUILTIN, PY_FROZEN, PKG_DIRECTORY
  71. BUILTIN_MODULE = C_BUILTIN
  72. FROZEN_MODULE = PY_FROZEN
  73.  
  74. class _Verbose:
  75.     
  76.     def __init__(self, verbose = VERBOSE):
  77.         self.verbose = verbose
  78.  
  79.     
  80.     def get_verbose(self):
  81.         return self.verbose
  82.  
  83.     
  84.     def set_verbose(self, verbose):
  85.         self.verbose = verbose
  86.  
  87.     
  88.     def note(self, *args):
  89.         if self.verbose:
  90.             self.message(*args)
  91.         
  92.  
  93.     
  94.     def message(self, format, *args):
  95.         if args:
  96.             print format % args
  97.         else:
  98.             print format
  99.  
  100.  
  101.  
  102. class BasicModuleLoader(_Verbose):
  103.     """Basic module loader.
  104.  
  105.     This provides the same functionality as built-in import.  It
  106.     doesn't deal with checking sys.modules -- all it provides is
  107.     find_module() and a load_module(), as well as find_module_in_dir()
  108.     which searches just one directory, and can be overridden by a
  109.     derived class to change the module search algorithm when the basic
  110.     dependency on sys.path is unchanged.
  111.  
  112.     The interface is a little more convenient than imp's:
  113.     find_module(name, [path]) returns None or 'stuff', and
  114.     load_module(name, stuff) loads the module.
  115.  
  116.     """
  117.     
  118.     def find_module(self, name, path = None):
  119.         if path is None:
  120.             path = [
  121.                 None] + self.default_path()
  122.         
  123.         for dir in path:
  124.             stuff = self.find_module_in_dir(name, dir)
  125.             if stuff:
  126.                 return stuff
  127.                 continue
  128.         
  129.  
  130.     
  131.     def default_path(self):
  132.         return sys.path
  133.  
  134.     
  135.     def find_module_in_dir(self, name, dir):
  136.         if dir is None:
  137.             return self.find_builtin_module(name)
  138.         else:
  139.             
  140.             try:
  141.                 return imp.find_module(name, [
  142.                     dir])
  143.             except ImportError:
  144.                 return None
  145.  
  146.  
  147.     
  148.     def find_builtin_module(self, name):
  149.         if imp.is_builtin(name):
  150.             return (None, '', ('', '', BUILTIN_MODULE))
  151.         
  152.         if imp.is_frozen(name):
  153.             return (None, '', ('', '', FROZEN_MODULE))
  154.         
  155.  
  156.     
  157.     def load_module(self, name, stuff):
  158.         (file, filename, info) = stuff
  159.         
  160.         try:
  161.             return imp.load_module(name, file, filename, info)
  162.         finally:
  163.             if file:
  164.                 file.close()
  165.             
  166.  
  167.  
  168.  
  169.  
  170. class Hooks(_Verbose):
  171.     '''Hooks into the filesystem and interpreter.
  172.  
  173.     By deriving a subclass you can redefine your filesystem interface,
  174.     e.g. to merge it with the URL space.
  175.  
  176.     This base class behaves just like the native filesystem.
  177.  
  178.     '''
  179.     
  180.     def get_suffixes(self):
  181.         return imp.get_suffixes()
  182.  
  183.     
  184.     def new_module(self, name):
  185.         return imp.new_module(name)
  186.  
  187.     
  188.     def is_builtin(self, name):
  189.         return imp.is_builtin(name)
  190.  
  191.     
  192.     def init_builtin(self, name):
  193.         return imp.init_builtin(name)
  194.  
  195.     
  196.     def is_frozen(self, name):
  197.         return imp.is_frozen(name)
  198.  
  199.     
  200.     def init_frozen(self, name):
  201.         return imp.init_frozen(name)
  202.  
  203.     
  204.     def get_frozen_object(self, name):
  205.         return imp.get_frozen_object(name)
  206.  
  207.     
  208.     def load_source(self, name, filename, file = None):
  209.         return imp.load_source(name, filename, file)
  210.  
  211.     
  212.     def load_compiled(self, name, filename, file = None):
  213.         return imp.load_compiled(name, filename, file)
  214.  
  215.     
  216.     def load_dynamic(self, name, filename, file = None):
  217.         return imp.load_dynamic(name, filename, file)
  218.  
  219.     
  220.     def load_package(self, name, filename, file = None):
  221.         return imp.load_module(name, file, filename, ('', '', PKG_DIRECTORY))
  222.  
  223.     
  224.     def add_module(self, name):
  225.         d = self.modules_dict()
  226.         if name in d:
  227.             return d[name]
  228.         
  229.         d[name] = m = self.new_module(name)
  230.         return m
  231.  
  232.     
  233.     def modules_dict(self):
  234.         return sys.modules
  235.  
  236.     
  237.     def default_path(self):
  238.         return sys.path
  239.  
  240.     
  241.     def path_split(self, x):
  242.         return os.path.split(x)
  243.  
  244.     
  245.     def path_join(self, x, y):
  246.         return os.path.join(x, y)
  247.  
  248.     
  249.     def path_isabs(self, x):
  250.         return os.path.isabs(x)
  251.  
  252.     
  253.     def path_exists(self, x):
  254.         return os.path.exists(x)
  255.  
  256.     
  257.     def path_isdir(self, x):
  258.         return os.path.isdir(x)
  259.  
  260.     
  261.     def path_isfile(self, x):
  262.         return os.path.isfile(x)
  263.  
  264.     
  265.     def path_islink(self, x):
  266.         return os.path.islink(x)
  267.  
  268.     
  269.     def openfile(self, *x):
  270.         return open(*x)
  271.  
  272.     openfile_error = IOError
  273.     
  274.     def listdir(self, x):
  275.         return os.listdir(x)
  276.  
  277.     listdir_error = os.error
  278.  
  279.  
  280. class ModuleLoader(BasicModuleLoader):
  281.     """Default module loader; uses file system hooks.
  282.  
  283.     By defining suitable hooks, you might be able to load modules from
  284.     other sources than the file system, e.g. from compressed or
  285.     encrypted files, tar files or (if you're brave!) URLs.
  286.  
  287.     """
  288.     
  289.     def __init__(self, hooks = None, verbose = VERBOSE):
  290.         BasicModuleLoader.__init__(self, verbose)
  291.         if not hooks:
  292.             pass
  293.         self.hooks = Hooks(verbose)
  294.  
  295.     
  296.     def default_path(self):
  297.         return self.hooks.default_path()
  298.  
  299.     
  300.     def modules_dict(self):
  301.         return self.hooks.modules_dict()
  302.  
  303.     
  304.     def get_hooks(self):
  305.         return self.hooks
  306.  
  307.     
  308.     def set_hooks(self, hooks):
  309.         self.hooks = hooks
  310.  
  311.     
  312.     def find_builtin_module(self, name):
  313.         if self.hooks.is_builtin(name):
  314.             return (None, '', ('', '', BUILTIN_MODULE))
  315.         
  316.         if self.hooks.is_frozen(name):
  317.             return (None, '', ('', '', FROZEN_MODULE))
  318.         
  319.  
  320.     
  321.     def find_module_in_dir(self, name, dir, allow_packages = 1):
  322.         if dir is None:
  323.             return self.find_builtin_module(name)
  324.         
  325.         if allow_packages:
  326.             fullname = self.hooks.path_join(dir, name)
  327.             if self.hooks.path_isdir(fullname):
  328.                 stuff = self.find_module_in_dir('__init__', fullname, 0)
  329.                 if stuff:
  330.                     file = stuff[0]
  331.                     if file:
  332.                         file.close()
  333.                     
  334.                     return (None, fullname, ('', '', PKG_DIRECTORY))
  335.                 
  336.             
  337.         
  338.         for info in self.hooks.get_suffixes():
  339.             (suff, mode, type) = info
  340.             fullname = self.hooks.path_join(dir, name + suff)
  341.             
  342.             try:
  343.                 fp = self.hooks.openfile(fullname, mode)
  344.                 return (fp, fullname, info)
  345.             continue
  346.             except self.hooks.openfile_error:
  347.                 continue
  348.             
  349.  
  350.         
  351.  
  352.     
  353.     def load_module(self, name, stuff):
  354.         (file, filename, info) = stuff
  355.         (suff, mode, type) = info
  356.         
  357.         try:
  358.             if type == BUILTIN_MODULE:
  359.                 return self.hooks.init_builtin(name)
  360.             
  361.             if type == FROZEN_MODULE:
  362.                 return self.hooks.init_frozen(name)
  363.             
  364.             if type == C_EXTENSION:
  365.                 m = self.hooks.load_dynamic(name, filename, file)
  366.             elif type == PY_SOURCE:
  367.                 m = self.hooks.load_source(name, filename, file)
  368.             elif type == PY_COMPILED:
  369.                 m = self.hooks.load_compiled(name, filename, file)
  370.             elif type == PKG_DIRECTORY:
  371.                 m = self.hooks.load_package(name, filename, file)
  372.             else:
  373.                 raise ImportError, 'Unrecognized module type (%r) for %s' % (type, name)
  374.         finally:
  375.             if file:
  376.                 file.close()
  377.             
  378.  
  379.         m.__file__ = filename
  380.         return m
  381.  
  382.  
  383.  
  384. class FancyModuleLoader(ModuleLoader):
  385.     '''Fancy module loader -- parses and execs the code itself.'''
  386.     
  387.     def load_module(self, name, stuff):
  388.         (suff, mode, type) = (file, filename)
  389.         realfilename = filename
  390.         path = None
  391.         if type == FROZEN_MODULE:
  392.             code = self.hooks.get_frozen_object(name)
  393.         elif type == PY_COMPILED:
  394.             import marshal as marshal
  395.             file.seek(8)
  396.             code = marshal.load(file)
  397.         elif type == PY_SOURCE:
  398.             data = file.read()
  399.             code = compile(data, realfilename, 'exec')
  400.         else:
  401.             return ModuleLoader.load_module(self, name, stuff)
  402.         m = self.hooks.add_module(name)
  403.         if path:
  404.             m.__path__ = path
  405.         
  406.         m.__file__ = filename
  407.         
  408.         try:
  409.             exec code in m.__dict__
  410.         except:
  411.             d = self.hooks.modules_dict()
  412.             if name in d:
  413.                 del d[name]
  414.             
  415.             raise 
  416.  
  417.         return m
  418.  
  419.  
  420.  
  421. class BasicModuleImporter(_Verbose):
  422.     '''Basic module importer; uses module loader.
  423.  
  424.     This provides basic import facilities but no package imports.
  425.  
  426.     '''
  427.     
  428.     def __init__(self, loader = None, verbose = VERBOSE):
  429.         _Verbose.__init__(self, verbose)
  430.         if not loader:
  431.             pass
  432.         self.loader = ModuleLoader(None, verbose)
  433.         self.modules = self.loader.modules_dict()
  434.  
  435.     
  436.     def get_loader(self):
  437.         return self.loader
  438.  
  439.     
  440.     def set_loader(self, loader):
  441.         self.loader = loader
  442.  
  443.     
  444.     def get_hooks(self):
  445.         return self.loader.get_hooks()
  446.  
  447.     
  448.     def set_hooks(self, hooks):
  449.         return self.loader.set_hooks(hooks)
  450.  
  451.     
  452.     def import_module(self, name, globals = { }, locals = { }, fromlist = []):
  453.         name = str(name)
  454.         if name in self.modules:
  455.             return self.modules[name]
  456.         
  457.         stuff = self.loader.find_module(name)
  458.         if not stuff:
  459.             raise ImportError, 'No module named %s' % name
  460.         
  461.         return self.loader.load_module(name, stuff)
  462.  
  463.     
  464.     def reload(self, module, path = None):
  465.         name = str(module.__name__)
  466.         stuff = self.loader.find_module(name, path)
  467.         if not stuff:
  468.             raise ImportError, 'Module %s not found for reload' % name
  469.         
  470.         return self.loader.load_module(name, stuff)
  471.  
  472.     
  473.     def unload(self, module):
  474.         del self.modules[str(module.__name__)]
  475.  
  476.     
  477.     def install(self):
  478.         self.save_import_module = __builtin__.__import__
  479.         self.save_reload = __builtin__.reload
  480.         if not hasattr(__builtin__, 'unload'):
  481.             __builtin__.unload = None
  482.         
  483.         self.save_unload = __builtin__.unload
  484.         __builtin__.__import__ = self.import_module
  485.         __builtin__.reload = self.reload
  486.         __builtin__.unload = self.unload
  487.  
  488.     
  489.     def uninstall(self):
  490.         __builtin__.__import__ = self.save_import_module
  491.         __builtin__.reload = self.save_reload
  492.         __builtin__.unload = self.save_unload
  493.         if not __builtin__.unload:
  494.             del __builtin__.unload
  495.         
  496.  
  497.  
  498.  
  499. class ModuleImporter(BasicModuleImporter):
  500.     '''A module importer that supports packages.'''
  501.     
  502.     def import_module(self, name, globals = None, locals = None, fromlist = None):
  503.         parent = self.determine_parent(globals)
  504.         (q, tail) = self.find_head_package(parent, str(name))
  505.         m = self.load_tail(q, tail)
  506.         if not fromlist:
  507.             return q
  508.         
  509.         if hasattr(m, '__path__'):
  510.             self.ensure_fromlist(m, fromlist)
  511.         
  512.         return m
  513.  
  514.     
  515.     def determine_parent(self, globals):
  516.         if not globals or '__name__' not in globals:
  517.             return None
  518.         
  519.         pname = globals['__name__']
  520.         if '__path__' in globals:
  521.             parent = self.modules[pname]
  522.             return parent
  523.         
  524.         if '.' in pname:
  525.             i = pname.rfind('.')
  526.             pname = pname[:i]
  527.             parent = self.modules[pname]
  528.             return parent
  529.         
  530.  
  531.     
  532.     def find_head_package(self, parent, name):
  533.         if '.' in name:
  534.             i = name.find('.')
  535.             head = name[:i]
  536.             tail = name[i + 1:]
  537.         else:
  538.             head = name
  539.             tail = ''
  540.         if parent:
  541.             qname = '%s.%s' % (parent.__name__, head)
  542.         else:
  543.             qname = head
  544.         q = self.import_it(head, qname, parent)
  545.         if q:
  546.             return (q, tail)
  547.         
  548.         if parent:
  549.             qname = head
  550.             parent = None
  551.             q = self.import_it(head, qname, parent)
  552.             if q:
  553.                 return (q, tail)
  554.             
  555.         
  556.         raise ImportError, 'No module named ' + qname
  557.  
  558.     
  559.     def load_tail(self, q, tail):
  560.         m = q
  561.         while tail:
  562.             i = tail.find('.')
  563.             if i < 0:
  564.                 i = len(tail)
  565.             
  566.             head = tail[:i]
  567.             tail = tail[i + 1:]
  568.             mname = '%s.%s' % (m.__name__, head)
  569.             m = self.import_it(head, mname, m)
  570.             if not m:
  571.                 raise ImportError, 'No module named ' + mname
  572.                 continue
  573.         return m
  574.  
  575.     
  576.     def ensure_fromlist(self, m, fromlist, recursive = 0):
  577.         for sub in fromlist:
  578.             if sub == '*':
  579.                 if not recursive:
  580.                     
  581.                     try:
  582.                         all = m.__all__
  583.                     except AttributeError:
  584.                         pass
  585.  
  586.                     self.ensure_fromlist(m, all, 1)
  587.                     continue
  588.                 continue
  589.             
  590.             if sub != '*' and not hasattr(m, sub):
  591.                 subname = '%s.%s' % (m.__name__, sub)
  592.                 submod = self.import_it(sub, subname, m)
  593.                 if not submod:
  594.                     raise ImportError, 'No module named ' + subname
  595.                 
  596.             submod
  597.         
  598.  
  599.     
  600.     def import_it(self, partname, fqname, parent, force_load = 0):
  601.         if not partname:
  602.             raise ValueError, 'Empty module name'
  603.         
  604.         if not force_load:
  605.             
  606.             try:
  607.                 return self.modules[fqname]
  608.             except KeyError:
  609.                 pass
  610.             except:
  611.                 None<EXCEPTION MATCH>KeyError
  612.             
  613.  
  614.         None<EXCEPTION MATCH>KeyError
  615.         
  616.         try:
  617.             if parent:
  618.                 pass
  619.             path = parent.__path__
  620.         except AttributeError:
  621.             return None
  622.  
  623.         partname = str(partname)
  624.         stuff = self.loader.find_module(partname, path)
  625.         if not stuff:
  626.             return None
  627.         
  628.         fqname = str(fqname)
  629.         m = self.loader.load_module(fqname, stuff)
  630.         if parent:
  631.             setattr(parent, partname, m)
  632.         
  633.         return m
  634.  
  635.     
  636.     def reload(self, module):
  637.         name = str(module.__name__)
  638.         if '.' not in name:
  639.             return self.import_it(name, name, None, force_load = 1)
  640.         
  641.         i = name.rfind('.')
  642.         pname = name[:i]
  643.         parent = self.modules[pname]
  644.         return self.import_it(name[i + 1:], name, parent, force_load = 1)
  645.  
  646.  
  647. default_importer = None
  648. current_importer = None
  649.  
  650. def install(importer = None):
  651.     global current_importer
  652.     if not importer and default_importer:
  653.         pass
  654.     current_importer = ModuleImporter()
  655.     current_importer.install()
  656.  
  657.  
  658. def uninstall():
  659.     current_importer.uninstall()
  660.  
  661.